package com.cg.leetcode;

import org.junit.Test;

/**
 * 202.快乐数
 *
 * @author cg
 * @program LeetCode->LeetCode_202
 * @create 2022-07-17 21:50
 **/
public class LeetCode_202 {

    @Test
    public void test202() {
        System.out.println(isHappy(19));
        System.out.println(isHappy(9));
    }

    /**
     * 编写一个算法来判断一个数 n 是不是快乐数。
     * 「快乐数」 定义为：
     * 对于一个正整数，每一次将该数替换为它每个位置上的数字的平方和。
     * 然后重复这个过程直到这个数变为 1，也可能是 无限循环 但始终变不到 1。
     * 如果这个过程 结果为 1，那么这个数就是快乐数。
     * 如果 n 是 快乐数 就返回 true ；不是，则返回 false 。
     * <p>
     * 示例 1：
     * 输入：n = 19
     * 输出：true
     * 解释：
     * 1^2 + 9^2 = 82
     * 8^2 + 2^2 = 68
     * 6^2 + 8^2 = 100
     * 1^2 + 0^2 + 0^2 = 1
     * <p>
     * 示例 2：
     * 输入：n = 2
     * 输出：false
     * <p>
     * 提示：
     * 1 <= n <= 2^31 - 1
     *
     * @param n
     * @return
     */
    public boolean isHappy(int n) {
        // 双指针
        int slow = n, fast = getNextNum(n);
        while (fast != 1 && fast != slow) {
            slow = getNextNum(slow);
            fast = getNextNum(getNextNum(fast));
        }
        return fast == 1;
    }

    /*public boolean isHappy(int n) {
        // Hash
        Set<Integer> seen = new HashSet<>();
        while (n != 1 && !seen.contains(n)) {
            seen.add(n);
            n = getNextNum(n);
        }
        return n == 1;
    }*/

    private int getNextNum(int n) {
        int res = 0;
        while (n > 0) {
            int temp = n % 10;
            res += temp * temp;
            n /= 10;
        }
        return res;
    }
}
